"""
代理迭代

问题：
    你构建了一个自定义容器对象，里面包含有列表、元组或其他可迭代对象。你想直接在你的这个新容器对象上执行迭代操作。
解决方案：
    实际上你只需要定义一个 __iter__() 方法，将迭代操作代理到容器内部的对象上去。
"""


class Node:
    def __init__(self, value):
        self._value = value
        self._children = []

    def add_child(self, node):
        self._children.append(node)

    def __iter__(self):
        return iter(self._children)

    def __str__(self):
        return str(self._value)


def use_delegate_iter():
    root = Node(0)
    child1 = Node(1)
    child2 = Node(2)
    root.add_child(child1)
    root.add_child(child2)

    for ch in root:
        print(ch)

    # 输出结果：
    """
    1
    2
    """


if __name__ == '__main__':
    use_delegate_iter()
